﻿#region Copyright & licence

// This file is part of Refix.
// 
// Refix is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
// 
// Refix is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
// 
// You should have received a copy of the GNU Lesser General Public
// License along with Refix.  If not, see <http://www.gnu.org/licenses/>.
// 
// Copyright (C) 2010-11 David Musgrove and others.

#endregion

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;

namespace Refix.Core.Structure
{
    public class Solution : IEnumerable<BinaryReference>
    {
        private readonly List<Project> _projects = new List<Project>();

        public List<Project> Projects
        {
            get { return _projects; }
        }

        public string FileName { get; set; }

        #region IEnumerable<BinaryReference> Members

        public IEnumerator<BinaryReference> GetEnumerator()
        {
            return (from project in Projects
                    from reference in project.References
                    from subReference in RecurseReferences(reference)
                    select subReference).GetEnumerator();
        }

        IEnumerator IEnumerable.GetEnumerator()
        {
            return GetEnumerator();
        }

        #endregion

        static private IEnumerable<BinaryReference> RecurseReferences(BinaryReference reference)
        {
            yield return reference;
            foreach (BinaryReference subReference in reference.References)
            {
                foreach (BinaryReference subSubReference in RecurseReferences(subReference))
                {
                    yield return subSubReference;
                }
            }
        }

        public void AcceptVisitor(IStructureVisitor visitor)
        {
            visitor.ProcessSolution(this);
            var positionAware = visitor as IPositionAwareStructureVisitor;
            foreach (Project project in Projects)
            {
                if (positionAware != null)
                {
                    positionAware.NextPosition = Projects.PositionOf(project);
                }
                visitor.Visit(project);
                if (positionAware != null)
                {
                    positionAware.NextPosition = Position.Normal;
                }
            }
            visitor.AfterProcessSolution();
        }
    }
}